package com.xpqh.ai.dto;

import lombok.Data;

/**
 * 聊天响应DTO
 */
@Data
public class ChatResponse {
    
    /**
     * 响应内容
     */
    private String content;
    
    /**
     * 使用的模型
     */
    private String model;
    
    /**
     * 消耗的token数
     */
    private Integer totalTokens;
    
    /**
     * 提示token数
     */
    private Integer promptTokens;
    
    /**
     * 完成token数
     */
    private Integer completionTokens;
    
    /**
     * 请求ID
     */
    private String requestId;
    
    /**
     * 处理时间（毫秒）
     */
    private Long processingTime;
    
    /**
     * 是否成功
     */
    private Boolean success = true;
    
    /**
     * 错误信息
     */
    private String errorMessage;
    
    /**
     * 创建成功响应
     */
    public static ChatResponse success(String content, String model) {
        ChatResponse response = new ChatResponse();
        response.setContent(content);
        response.setModel(model);
        response.setSuccess(true);
        return response;
    }
    
    /**
     * 创建错误响应
     */
    public static ChatResponse error(String errorMessage) {
        ChatResponse response = new ChatResponse();
        response.setSuccess(false);
        response.setErrorMessage(errorMessage);
        return response;
    }
} 